package com.lark.algorithm.train.onChapter.linkedList;

/**
 * @author btmood
 * @version 1.0
 * @apiNote
 * Q203:移除链表元素
 * 给你一个链表的头节点 head 和一个整数 val ，请你删除链表中所有满足 Node.val == val 的节点，并返回 新的头节点 。
 *
 * 示例 1：
 *
 *
 * 输入：head = [1,2,6,3,4,5,6], val = 6
 * 输出：[1,2,3,4,5]
 * 示例 2：
 *
 * 输入：head = [], val = 1
 * 输出：[]
 * 示例 3：
 *
 * 输入：head = [7,7,7,7], val = 7
 * 输出：[]
 *
 *
 * 提示：
 *
 * 列表中的节点数目在范围 [0, 104] 内
 * 1 <= Node.val <= 50
 * 0 <= val <= 50
 */
public class D20240516Q203_移除链表元素 {


    /**
     * 添加虚拟头节点
     * @param head
     * @param val
     * @return
     */
    public ListNode removeElements(ListNode head, int val) {
        // 添加虚拟头节点，方便删除头节点的操作
        ListNode dummyHead = new ListNode(val, head);
        // 待判断的节点的前一个节点
        ListNode prevNode = dummyHead;
        while (prevNode.next != null) {
            // 待判断的节点
            ListNode retNode = prevNode.next;
            // 待判断节点的值和val相同
            if (retNode.val == val) {
                prevNode.next = retNode.next;
                retNode.next = null;
            } else {
                prevNode = prevNode.next;
            }
        }
        return dummyHead.next;
    }

    /**
     * 递归解法
     * @param head
     * @param val
     * @return
     */
    public ListNode removeElements1(ListNode head, int val) {
        if (head == null) {
            return null;
        }
        head.next = removeElements1(head.next, val);
        if (head.val == val) {
            return head.next;
        } else {
            return head;
        }
    }

    public static class ListNode {
        int val;
        ListNode() {}
        ListNode(int val) { this.val = val; }
        ListNode(int val, ListNode next) { this.val = val; this.next = next; }
        ListNode next;
    }

}


